跳到主要内容

GPU 型号与计算能力

在使用 Vllm 时,使用 V100 显卡和 A100 显卡会显示,V100 不支持 BF16 精度。那么具体的 GPU 型号和支持精度的关系是怎么样的?

Hardware and Precision

CUDA Compute CapabilityExample DevicesTF32FP32FP16FP8FP4BF16INT8FP16 Tensor CoresINT8 Tensor CoresDLA
12.0NVIDIA RTX 5090YesYesYesYesYesYesYesYesYesNo
10.0NVIDIA B200YesYesYesYesYesYesYesYesYesNo
9.0NVIDIA H100YesYesYesYesYes 5YesYesYesYesNo
9.0NVIDIA GH200 480 GBYesYesYesYesYes 5YesYesYesYesNo
8.9NVIDIA L40SYesYesYesYesYes 5YesYesYesYesNo
8.7NVIDIA DRIVE AGX OrinYesYesYesNoNoNoYesYesYesYes
8.6NVIDIA A10YesYesYesNoNoYesYesYesYesNo
8.0NVIDIA A100YesYesYesNoNoYesYesYesYesNo
7.5NVIDIA T4NoYesYesNoNoNoYesYesYesNo

GPU Compute Capability

GPUCompute Capability
NVIDIA H1009.0
NVIDIA L48.9
NVIDIA L408.9
NVIDIA A1008.0
NVIDIA A408.6
NVIDIA A308.0
NVIDIA A108.6
NVIDIA A168.6
NVIDIA A28.6
NVIDIA T47.5
NVIDIA V1007.0
Tesla P1006.0
Tesla P406.1
Tesla P46.1
Tesla M605.2
Tesla M405.2
Tesla K803.7
Tesla K403.5
Tesla K203.5
Tesla K103.0